KV$^2$: A Self-Refining KV Cache
KV²:一种自精炼 KV 缓存压缩方法
🧭 LLM 推理 · KV cache 压缩 ★ 深度解读
Karlsruhe Institute of Technology
用轻量代理选少量锚点做选择性重建,在 2% KV 预算下大幅优于 KVzip 等基线
📌 工作参考价值(4 点)
- 对 Agent 长期记忆、用户历史、商品知识库等“一次缓存、多次 query”场景,优先用 query-agnostic 压缩;KV² 两阶段思路可复制:先用便宜业务信号(如 key 相似度、商品/行为 embedding)挑锚点,再做局部 max-attention 打分,比全量重建省钱。
- 可将 KV 预算压到 2%–5%,适合电商搜索/推荐的超长上下文(长用户序列、多商品文档),降低缓存显存和 decode 带宽;chunk size 4K 是性价比甜点。
- 默认单轮即可,别急着上多轮 self-refinement;只有在 proxy 很弱或 reconstruction 比例极小时才考虑 2–3 轮。
- 落地注意:论文基于 kvpress 的 masking 实现未物理 compact,需配合 headwise 变长 attention、paged cache 和重复 query 服务后端,才能把压缩阶段收益转成实际 decode 收益。